package com.ruoyi.generator.service;

import org.springframework.stereotype.Service;

import java.util.concurrent.ConcurrentHashMap;

@Service
public class PrepareHtmlDataHelper {

    private ConcurrentHashMap<String, IPrepareHtmlDataService> prepareHtmlDataServiceMap = new ConcurrentHashMap<>();

    public void putPrepareHtmlDataService(String pageType, IPrepareHtmlDataService prepareHtmlDataService) {
        prepareHtmlDataServiceMap.put(pageType, prepareHtmlDataService);
    }

    public IPrepareHtmlDataService getPrepareHtmlDataService(String pageType) {
        return prepareHtmlDataServiceMap.get(pageType);
    }

}
